package com.zqf.shiro.filter;



import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.zqf.shiro.pojo.LoginUser;
import com.zqf.shiro.utils.JwtHelper;
import com.zqf.shiro.utils.LoginUserUtils;
import com.zqf.shiro.utils.SpringContextUtils;
import org.apache.shiro.web.filter.authc.BasicHttpAuthenticationFilter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.redis.core.RedisTemplate;

import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import java.io.Serializable;


public class JwtFilter extends BasicHttpAuthenticationFilter {


    private final RedisTemplate<String, Serializable> redisTemplate;


    private static final Logger logger = LoggerFactory.getLogger(JwtFilter.class);

    public JwtFilter() {
        redisTemplate = SpringContextUtils.getBean("ddbRedisTemplate", RedisTemplate.class);
    }

    /**
     * 判断用户是否想要登入。
     * 检测header里面是否包含Authorization字段即可
     */
    @Override
    protected boolean isLoginAttempt(ServletRequest request, ServletResponse response) {
        String authorization = super.getAuthzHeader(request);
        LoginUser loginUser;
        if (StringUtils.isBlank(authorization) || (loginUser = JwtHelper.checkAndGetUser(authorization)) == null) {
            loginUser = new LoginUser();

        }
        LoginUserUtils.setUser(loginUser);
        return true;
    }

    /**
     *登录
     */
    @Override
    protected boolean executeLogin(ServletRequest request, ServletResponse response) {
        String authorization = super.getAuthzHeader(request);
        LoginUser loginUser = JwtHelper.checkAndGetUser(authorization);
        if (loginUser == null) {
            return false;
        }
        JwtToken token = new JwtToken(authorization.replaceFirst("Bearer ", ""));
        // 提交给realm进行登入，如果错误他会抛出异常并被捕获
        getSubject(request, response).login(token);

        logger.debug("登录成功");
        // 如果没有抛出异常则代表登入成功，返回true
        return true;
    }

    /**
     * 这里我们详细说明下为什么最终返回的都是true，即允许访问
     * 例如我们提供一个地址 GET /article
     * 登入用户和游客看到的内容是不同的
     * 如果在这里返回了false，请求会被直接拦截，用户看不到任何东西
     * 所以我们在这里返回true，Controller中可以通过 subject.isAuthenticated() 来判断用户是否登入
     * 如果有些资源只有登入用户才能访问，我们只需要在方法上面加上 @RequiresAuthentication 注解即可
     * 但是这样做有一个缺点，就是不能够对GET,POST等请求进行分别过滤鉴权(因为我们重写了官方的方法)，但实际上对应用影响不大
     */
    @Override
    protected boolean isAccessAllowed(ServletRequest request, ServletResponse response, Object mappedValue) {
        if (isLoginAttempt(request, response)) {
            try {
                executeLogin(request, response);
            } catch (Exception e) {
                logger.error("登录错误:", e);
                return false;
            }
        }
        return true;
    }

    /**
     * 操作完成后清除登录信息
     */
    @Override
    public void afterCompletion(ServletRequest request, ServletResponse response, Exception exception) throws Exception {
        logger.debug("清除用户信息");
        super.afterCompletion(request, response, exception);
        LoginUserUtils.clear();
    }
}